from argparse import Namespace
import torch

config = Namespace(

    AVAIL_GPUS = min(1, torch.cuda.device_count()),
    BATCH_SIZE = 64 if min(1, torch.cuda.device_count()) else 64,
    lr = 0.02,
    optim_type = "Adam",
    max_epochs = 3,
    ckpt_path = "best.ckpt",
    data_dir ="../../../../data/pokeman",
    dropout_p = 0.1,
    modelPATH = "modelPATH",

)


